package com.itheima.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.itheima.reggie.common.R;
import com.itheima.reggie.dto.SetmealDto;
import com.itheima.reggie.entity.Category;
import com.itheima.reggie.entity.Dish;
import com.itheima.reggie.entity.Setmeal;
import com.itheima.reggie.service.CategoryService;
import com.itheima.reggie.service.DishService;
import com.itheima.reggie.service.SetmealService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.io.Serializable;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author Mylinja 2021-12-19 星期日 21:19
 */
@Slf4j
@RestController
@RequestMapping("/setmeal")
public class SetmealController {

    @Autowired
    private SetmealService setmealService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private DishService dishService;

    /**
     * 新增套餐方法
     *
     * @param setmealDto 套餐拓展类
     * @return 返回结果
     */
    @PostMapping
    public R<String> save(@RequestBody SetmealDto setmealDto) {
        log.info("新增套餐信息：{}", setmealDto);

        setmealService.saveWithDish(setmealDto);
        return R.success("添加成功");
    }

    /**
     * 分页查询所有套餐方法
     *
     * @param page     第几页
     * @param pageSize 每页多少条
     * @param name     查询条件
     * @return 返回结果
     */
    @GetMapping("/page")
    @SuppressWarnings({"unchecked", "varargs"})
    public R<Page<SetmealDto>> page(int page, int pageSize, String name) {
        log.info("开始查询所有套餐信息...");

        //1 构建分页器
        Page<Setmeal> pageInfo = new Page<>(page, pageSize);
        Page<SetmealDto> dtoPage = new Page<>();

        //2 构建查询器
        LambdaQueryWrapper<Setmeal> query = new LambdaQueryWrapper<>();
        query.like(name != null, Setmeal::getName, name);
        query.orderByDesc(Setmeal::getUpdateTime);

        //3 查询，将pageInfo拷贝给dtoPage
        setmealService.page(pageInfo, query);
        BeanUtils.copyProperties(pageInfo, dtoPage);

        //4 把pageInfo里的records集合，加工一下，赋值给dtoPage中的records集合
        List<Setmeal> records = pageInfo.getRecords();

        List<SetmealDto> collect = records.stream().map(setmeal -> {
            SetmealDto setmealDto = new SetmealDto();
            // 把setmeal对象，拷贝到setmealDto中
            BeanUtils.copyProperties(setmeal, setmealDto);

            // 根据categoryId，设置setmealDto对象的CategoryName属性
            Long categoryId = setmeal.getCategoryId();
            Category byId = categoryService.getById(categoryId);
            if (byId != null) {
                setmealDto.setCategoryName(byId.getName());
            }
            return setmealDto;
        }).collect(Collectors.toList());

        //5 封装list集合并返回
        dtoPage.setRecords(collect);
        return R.success(dtoPage);
    }

    /**
     * 根据分类id，查询此分类下的所有套餐
     *
     * @param setmeal 套餐
     * @return 返回该分类id下的所有套餐集合
     */
    @GetMapping("/list")
    @SuppressWarnings({"unchecked", "varargs"})
    public R<List<Setmeal>> selectList(Setmeal setmeal) {
        log.info("查询所有套餐...");

        // 查询此套餐分类下的所有套餐
        LambdaQueryWrapper<Setmeal> query = new LambdaQueryWrapper<>();
        query.eq(setmeal.getCategoryId() != null, Setmeal::getCategoryId, setmeal.getCategoryId());
        // 只有起售状态的套餐才展示
        query.eq(Setmeal::getStatus, 1);
        query.orderByDesc(Setmeal::getUpdateTime);
        List<Setmeal> list = setmealService.list(query);

        return R.success(list);
    }

    /**
     * 删除套餐
     *
     * @param ids 套餐id集合
     * @return 返回结果
     */
    @DeleteMapping
    public R<String> delete(@RequestParam List<Long> ids) {
        log.info("ids：{}", ids);

        setmealService.removeWithDish(ids);
        return R.success("删除成功");
    }

    /**
     * 修改套餐状态
     *
     * @param state 状态码
     * @param ids   修改的集合
     * @return 返回结果
     */
    @PostMapping("/status/{state}")
    public R<String> update(@PathVariable Integer state, @RequestParam List<Long> ids) {
        log.info("批量修改套餐状态。。。");

        //1 根据ids修改套餐状态
        //SQL: update from setmeal set(status) value(0) where id in(ids)
        LambdaUpdateWrapper<Setmeal> query = new LambdaUpdateWrapper<>();
        query.in(Setmeal::getId, ids);
        query.set(Setmeal::getStatus, state);
        setmealService.update(query);

        return R.success("更新成功");
    }

    /**
     * 套餐信息回显
     *
     * @param id 修改的id
     * @return 返回结果
     */
    @GetMapping("/{id}")
    public R<SetmealDto> echoDtoInfo(@PathVariable long id) {
        log.info("开始回显套餐信息...");

        SetmealDto setmealDto = setmealService.getByIdWithDish(id);
        return R.success(setmealDto);
    }

    /**
     * 修改套餐方法
     *
     * @param setmealDto 套餐dto对象
     * @return 返回结果
     */
    @PutMapping
    public R<String> update(@RequestBody SetmealDto setmealDto) {
        log.info("执行更新套餐方法");

        setmealService.updateByIdWithDish(setmealDto);
        return R.success("修改成功");
    }

    /**
     * 前端页面用来展示菜品或套餐详情的方法
     * @param id 菜品或套餐id
     * @return 返回对象
     */
    @GetMapping("/dish/{id}")
    public R<? extends Serializable> showDish(@PathVariable Long id) {
        // 这个id不知道是菜品id还是套餐id
        // 如果是dish_id
        Dish dish = dishService.getById(id);
        if (dish != null) {
            return R.success(dish);
        }

        // 那么id是setmealId
        Setmeal setmeal = setmealService.getById(id);
        return R.success(setmeal);
    }
}
